package twelfth_day;

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;

import static java.util.Arrays.sort;

/**
 * 给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
 *
 * 请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。
 *
 * 注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，
 * 其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。
 */
public class MergeOrderedArray {
    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        if (m == 0) {
            System.arraycopy(nums2, 0, nums1, 0, n);
            System.out.println(Arrays.toString(nums1));
            return;
        }
        if (n == 0) {
            System.out.println(Arrays.toString(nums1));
            return;
        }

        // 直接合并后排序

        // 1
//        System.arraycopy(nums2, 0, nums1, m, m + n - 1);
//        sort(nums1);

        // 2
//        for (int i = m; i < m + n; i++) {
//            nums1[i] = nums2[i - m];
//        }
//        Arrays.sort(nums1);

        // 使用单调栈来解决
//        Deque<Integer> deque = new ArrayDeque<>();
//        int j = 0;
//        for (int i = 0; i < m; i++) {
//            int num1 = nums1[i];
//            while (j < n && num1 >= nums2[j]) {
//                int num2 = nums2[j ++];
//                // 当num1大于num2时将num2入栈
//                deque.push(num2);
//            }
//            // 已经将比num1小的全部入栈
//            deque.push(num1);
//        }
//        // 现在就只剩较大的
//        for (int i = j; i < n; i++) {
//            deque.push(nums2[i]);
//        }
//        for (int i = m + n - 1; i >= 0; --i) {
//            nums1[i] = deque.pop();
//        }

        // 使用双指针解决
        int[] sorted = new int[m + n];
        int flag1 = 0, flag2 = 0;
        int i = 0; // 控制sorted的
        // 总共只遍历了m + n
        while (flag1 < m || flag2 < n) {
            if (flag1 == m) {
                sorted[i ++] = nums2[flag2 ++];
            } else if (flag2 == n) {
                sorted[i ++] = nums1[flag1 ++];
            } else if (nums1[flag1] <= nums2[flag2]) {
                sorted[i ++] = nums1[flag1 ++];
            } else {
                sorted[i ++] = nums2[flag2 ++];
            }
        }
        System.arraycopy(sorted, 0, nums1, 0, m + n);
        System.out.println(Arrays.toString(nums1));
    }

    public static void main(String[] args) {
        //[4,0,0,0,0,0]
        //1
        //[1,2,3,5,6]
        //5
        int m = 1;
        int n = 2;
        int[] num1 = new int[] {0, 0, 0};
        int[] num2 = new int[] {1, 2};
        merge(num1, m, num2, n);
    }
}
